import os
def dirsize(d):#定义目录大小判断函数
    sum = 0
    if os.path.isfile(d):#判断文件大小功能
        sum+=os.path.getsize(d)
    if os.path.isdir(d):#判断目录大小功能
        dir_list = os.listdir(d)
        for i in dir_list:
            file = os.path.join(d,i)
            if os.path.isdir(file):
                sum+=dirsize(file)#递归
            if os.path.isfile(file):
                sum+=os.path.getsize(file)

    return sum
    #判断循环
while True:
    print("{0:5}{1:5}".format('1.输入文件目录','2.退出'))
    key = input("请输入对应选择")

    if key == "1":
        add_r = input("请输入目录或文件地址：")
        dirsize(add_r)
        print("{}字节".format(dirsize(add_r)))
        input("按回车键继续")
    elif key == "2":
        print("已推出")
        break
    else:
        print("输入有误")
